import argparse
import sys
from tx_hospitalgpt_w2v import hosipital_gpt

def args_from_cmdline():
    parser = argparse.ArgumentParser()
    parser.add_argument('--text', type=str, default=None)
    parser.add_argument('-w2v', '--w2v_path', type=str, default=r'D:\软工项目_TX智慧问诊\model\notebooks\Tencent_AILab_ChineseEmbedding.bin', help='w2v预训练词典路径，一般为txt，如果已经嵌入，可以直接输入bin文件路径。')
    parser.add_argument('-lstm', '--lstm_path', type=str, default=r'D:\软工项目_TX智慧问诊\model\notebooks\results\results\w2v_old\bilstm_model.pth')
    args = parser.parse_args()
    return args

if __name__ == "__main__":
    args=args_from_cmdline()

    if args.text == None:
        print('请输入症状描述。')
    else:
        medical_result_dialog, reply, medical_result, department_out, flag_call_120 = hosipital_gpt(args.text, args.w2v_path, args.lstm_path)

        print(medical_result_dialog)
        print(reply)

        sys.exit()


        '''if flag_call_120:
            print('您可以拨打120进行急救，有更多症状请继续描述。')'''
